package apptoko.com.adapter;

import java.util.LinkedHashMap;
import java.util.Map;

import android.content.Context;
import android.view.View;
import android.view.ViewGroup;
import android.widget.Adapter;
import android.widget.ArrayAdapter;
import android.widget.BaseAdapter;

import apptoko.com.R;

;
@SuppressWarnings("rawtypes")
public class SectionListAdapter extends BaseAdapter {

	
	public Map<String, ArrayAdapter> sections = new LinkedHashMap<String, ArrayAdapter>();
	public ArrayAdapter<String> headers;
	public static int TYPE_SECTION_HEADER = 0;
	
	public SectionListAdapter(Context context, int type){
		if(type == 1){
			headers = new ArrayAdapter<String>(context, R.layout.section_list);
		} else if(type == 2){
			headers = new ArrayAdapter<String>(context, R.layout.cate_section_list);
		}
	}
	
	public void addSection(String section, ArrayAdapter adapter)
	{
		this.headers.add(section);
		this.sections.put(section, adapter);
	}

	public void removeSection(String section){
		this.headers.remove(section);
		this.sections.remove(section);
	}

	@Override
	public int getCount() {
		int total = 0;
		for (ArrayAdapter adapter : this.sections.values())
			total += adapter.getCount() + 1;
		return total;
	}

	@Override
	public Object getItem(int position) {
		for (Object section : this.sections.keySet())
		{
			ArrayAdapter adapter = sections.get(section);
			int size = adapter.getCount() + 1;
			// check if position inside this section
			if (position == 0) return section;
			if (position < size) return adapter.getItem(position - 1);

			// otherwise jump into next section
			position -= size;
		}
	return null;
	}

	@Override
	public long getItemId(int position) {
		// TODO Auto-generated method stub
		return position;
	}

	@Override
	public View getView(int position, View convertView, ViewGroup parent) {
		int sectionnum = 0;
		for (Object section : this.sections.keySet())
			{
				Adapter adapter = sections.get(section);
				int size = adapter.getCount() + 1;

				// check if position inside this section
				if (position == 0) return headers.getView(sectionnum, convertView, parent);
				if (position < size) return adapter.getView(position - 1, convertView, parent);

				// otherwise jump into next section
				position -= size;
				sectionnum++;
			}
		return null;
	}


	@Override
	public int getViewTypeCount() {
		// assume that headers count as one, then total all sections
		int total = 1;
		for (Adapter adapter : this.sections.values())
			total += adapter.getViewTypeCount();
		return total;
	}


	@Override
	public int getItemViewType(int position) {
		int type = 1;
		for (Object section : this.sections.keySet())
			{
				Adapter adapter = sections.get(section);
				int size = adapter.getCount() + 1;

				// check if position inside this section
				if (position == 0) return TYPE_SECTION_HEADER;
				if (position < size) return type + adapter.getItemViewType(position - 1);

				// otherwise jump into next section
				position -= size;
				type += adapter.getViewTypeCount();
			}
		return -1;
	}

}
